import numpy as np 

def main():
    np.random.seed(1)
    X = np.random.randint(1, 10, size=30)
    # 将X处理为3列矩阵
    X2=X.reshape(-1,3)
    print("改为3列矩阵：\n",X2)

    # 将第三列中，小于等于3的修改为0、大于3且小于等于6的修改为1、大于6的修改为2
    arr=X2[:,2:]
    arr[arr<=3]=0
    arr[(arr>3) & (arr<=6)]=1
    arr[arr>6]=2
    print("数值修改后\n",X2)

    X_train=X2[:,0:2].copy()
    y_train=X2[:,2:].copy()

    # 通过numpy比较运算，通过y_train中的数据，分离出X_train中的三类数据
    for i in [0,1,2]:
        arr=y_train==i
        col=arr.reshape(-1)
        arr=X_train[col,:]
        print("第",(i+1),"个样本数据\n",arr)

if __name__=="__main__":
   main()